package com.lishem.carl._10dp;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * https://leetcode.cn/problems/word-break/description/
 * <p>
 * 给你一个字符串s和一个字符串列表wordDict作为字典。
 * <p>
 * 如果可以利用字典中出现的一个或多个单词拼接出 s 则返回 true。
 * <p>
 * 注意：不要求字典中出现的单词全部都使用，并且字典中的单词可以重复使用。
 * <p>
 * 示例 1：
 * <p>
 * 输入: s = "leetcode", wordDict = ["leet", "code"]
 * <p>
 * 输出: true
 * <p>
 * 解释: 返回 true 因为 "leetcode" 可以由 "leet" 和 "code" 拼接成。
 * <p>
 * 示例 2：
 * <p>
 * 输入: s = "applepenapple", wordDict = ["apple", "pen"]
 * <p>
 * 输出: true
 * <p>
 * 解释: 返回 true 因为 "applepenapple" 可以由 "apple" "pen" "apple" 拼接成。
 * <p>
 * 注意，你可以重复使用字典中的单词。
 * <p>
 * 示例 3：
 * <p>
 * 输入: s = "catsandog", wordDict = ["cats", "dog", "sand", "and", "cat"]
 * <p>
 * 输出: false
 */
public class _18LetCode139_单词拆分 {

    /**
     * 完全背包
     */
    public boolean wordBreak1(String s, List<String> wordDict) {
        Set<String> set = new HashSet<>(wordDict);
        boolean[] dp = new boolean[s.length() + 1];
        dp[0] = true;
        // 先遍历背包
        for (int i = 1; i <= s.length(); i++) {
            for (int j = 0; j < i; j++) {
                // substring 第一个是begin，第二个是end
                // 这个判断就是说，截取出来这个单词 在字典中
                // 并且除去这个单词依然能组成，那个加上这个单词也能组成
                if (set.contains(s.substring(j, i)) && dp[j]) {
                    dp[i] = true;
                }
                // 这个位置确定，就不需要继续判断了
                if (dp[i] == true) {
                    break;
                }
            }
        }
        return dp[s.length()];
    }

    /**
     * 思路一样，写法不一样
     */
    public boolean wordBreak2(String s, List<String> wordDict) {
        boolean[] dp = new boolean[s.length() + 1];
        dp[0] = true;
        // 先遍历背包
        for (int i = 1; i <= s.length(); i++) {
            // 再遍历物品
            for (String word : wordDict) {
                int len = word.length();
                // 够减  减去后前面一个位置能凑成   减完后中间这部分字符串和word相等
                if (i >= len && dp[i - len] && word.equals(s.substring(i - len, i))) {
                    dp[i] = true;
                    break;
                }
            }
        }
        return dp[s.length()];
    }


    public static void main(String[] args) {
        _18LetCode139_单词拆分 sol = new _18LetCode139_单词拆分();
        List<String> list = new ArrayList<>();
        list.add("leet");
        list.add("code");
        System.out.println(sol.wordBreak1("leetcode", list));
        System.out.println(sol.wordBreak2("leetcode", list));
        list = new ArrayList<>();
        list.add("apple");
        list.add("pen");
        System.out.println(sol.wordBreak1("applepenapple", list));
        System.out.println(sol.wordBreak2("applepenapple", list));
        list = new ArrayList<>();
        list.add("cats");
        list.add("dog");
        list.add("sand");
        list.add("and");
        list.add("cat");
        System.out.println(sol.wordBreak1("catsandog", list));
        System.out.println(sol.wordBreak2("catsandog", list));

    }
}
